package com.shiyanlou.upms.common.config;

import com.shiyanlou.upms.common.utils.R;
import org.springframework.boot.web.servlet.error.ErrorController;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * 统一异常处理(自定义异常处理机制)
 * 根据不同的错误类型，返回给用户不同的返回信息。Spring Boot 就提供了一个这样统一处理异常的接口类 ErrorController，并通过重写 getErrorPath() 方法实现异常页面跳转。
 */
@RestController
public class MainsiteErrorController implements ErrorController {

    private final String ERROR_PATH ="/error";  //错误页面标识

    /**
     * 出现错误，跳转到如下映射中
     * @return
     *
     * @Override注解 伪代码，表示子类重写父类的方法。不写也是可以的
     * 由于接口 ErrorController中没有此方法，报错，先省略
     */
    //@Override
    public String getErrorPath() {
        return ERROR_PATH;
    }
    /**
     * Web 页面错误处理
     */
    @RequestMapping(value = ERROR_PATH, produces = {"text/html"})   ////请求连接错误时，跳到相关文本html，值为/error页面
    public ModelAndView errorHtml(HttpServletRequest request, HttpServletResponse response) {
        int code = response.getStatus();    //获取相应状态码
        if (404 == code) {
            return new ModelAndView("error/404");
        } else if (403 == code) {
            return new ModelAndView("error/403");
        } else if (401 == code) {   //401状态码(unauthorized未授权)
            return new ModelAndView("login");
        } else {
            return new ModelAndView("error/500");
        }
    }

    @RequestMapping(value =ERROR_PATH)
    public R handleError(HttpServletRequest request, HttpServletResponse response) {
        int code = response.getStatus();
        if (404 == code) {
            return R.error(404, "未找到资源");
        } else if (403 == code) {
            return R.error(403, "没有访问权限");
        } else if (401 == code) {
            return R.error(401, "登录过期");
        } else {
            return R.error(500, "服务器错误");
        }
    }
}